/**
 * 代号:隐无为 2017：厚溥
 * 文件名：MyHandler.java
 * 创建人：柯栋
 * 日期：2017年12月19日
 * 修改人：
 * 描述：
 */
package org.kd.websocket;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.log4j.Logger;
import org.kd.model.Message;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketMessage;
import org.springframework.web.socket.WebSocketSession;

import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * 用途：消息处理类
 */
public class LayimChat implements WebSocketHandler {
	Logger log = Logger.getLogger(this.getClass());
	private static final List<WebSocketSession> users = new ArrayList<>();
	public static final Map<String, WebSocketSession> userSocketSessionMap = new HashMap<String, WebSocketSession>();
	// 用户进入系统监听 当新连接建立的时候，被调用

	@Override
	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
		users.add(session);
		log.info("id:" + session.getId() + "->用户连接成功");
	}

	// 处理消息 接受前端发送过来的消息
	@Override
	public void handleMessage(WebSocketSession session, WebSocketMessage<?> data) throws Exception {
		// 转换器
		ObjectMapper mapper = new ObjectMapper();

		// 将消息进行转化，因为是消息是json数据，可能里面包含了发送给某个人的信息，所以需要用json相关的工具类处理之后再封装成TextMessage，
		// 消息的封装格式一般有{from:xxxx,to:xxxxx,msg:xxxxx}，来自哪里，发送给谁，什么消息等等
		String messageJson = data.getPayload().toString();
		Message message = mapper.readValue(messageJson, Message.class);
		// 把消息包装成textMessage
		TextMessage textMessage = new TextMessage(message.getContent());
		log.info(textMessage);
		// 给所有用户群发消息
		// sendMessagesToUsers(textMessage);
		// 给当前WebSocketSession发
		session.sendMessage(textMessage);
		// 给指定用户
		// sendMessageToUser2("666",new TextMessage("kkkk666"));

	}

	// 后台错误信息处理方法
	@Override
	public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {

	}

	// 用户退出后的处理，不如退出之后，要将用户信息从websocket的session中remove掉，这样用户就处于离线状态了，也不会占用系统资源
	@Override
	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
		if (session.isOpen()) {
			session.close();
		}
		users.remove(session);
		log.info("退出系统");

	}

	@Override
	public boolean supportsPartialMessages() {
		return false;
	}

	/**
	 * 给所有的用户发送消息
	 */
	public void sendMessagesToUsers(TextMessage message) {
		for (WebSocketSession user : users) {
			try {
				// isOpen()在线就发送
				if (user.isOpen()) {
					user.sendMessage(message);
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 * 给某个用户发送消息
	 * 
	 * @param userid
	 * @param message
	 * @throws IOException
	 */
	public void sendMessageToUser2(String userid, TextMessage message) throws IOException {
		WebSocketSession session = userSocketSessionMap.get(userid);
		log.info(message);
		if (session != null && session.isOpen()) {
			session.sendMessage(message);
		} else {
			log.info("用户：" + userid + "websocketSession已失效！");
		}
	}

	/**
	 * 发送消息给指定的用户
	 */
	public void sendMessageToUser(String userId, TextMessage message) {
		for (WebSocketSession user : users) {
			if (user.getAttributes().get("666").equals(userId)) {
				try {
					// isOpen()在线就发送
					if (user.isOpen()) {
						user.sendMessage(message);
					}
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

}
